# 1.导入模块
from openpyxl import load_workbook

# 2.加载 excel 文件
wb = load_workbook('01今日访客信息.xlsx')

# 3.获取默认表
sheet = wb.active

data_list = []
# 4.循环遍历 获取 所有 表数据
# 1. 循环 所有  rows
for row in sheet.rows:
    # 2. 循环 每行 中 单元格 cell
    data = []
    for cell in row:
        # 3. 获取单元格的数据  values
        val = cell.value
        # 4. 放入 这一行 list
        data.append(val)
    # 5. 再将 一行 数据--放入 到最大列表
    data_list.append(data)

# 5. 删除表头 数据
del data_list[0]
print(data_list)

# 6.写入到mysql数据
# 6.1 导入模块
import pymysql

# 6.2 实例化对象 --连接mysql的数据库
connection = pymysql.connect(
    host='localhost',
    user='root',
    password='mysql',
    db='car',
    charset='utf8',
    cursorclass=pymysql.cursors.DictCursor
)

# 6.3 使用 cursor对象
with connection.cursor() as cursor:
    for data in data_list:
        # 6.4 增加数据   sql
        sql = "insert into car_custom(name, phone) values(%s, %s);"
        # 6.5 cursor 执行 语句
        cursor.execute(sql, data)

# 6.6 提交数据  ---mysql
connection.commit()

# 6.7 关闭链接
connection.close()

print('写入mysql成功!')
